package com.sfzd5.fasttextknowledge;

import com.github.jfasttext.JFastText;

import java.io.File;
import java.util.List;

public class Test {
    public static void main(String[] args) {
        JFastText jft = new JFastText();
        File model = new File("mode", "ls.model");
        jft.loadModel(model.getAbsolutePath());

        //jft.loadModel("ls.model");
        String text = "3.周杰伦歌曲《爱在西元前》的歌词写道：“古巴比伦王颁布了……，刻在黑色的玄武岩，距今已经三千七百多年。”歌词所唱“刻在黑色玄武岩，距今已经三千七百多年”的文献是() A.《汉谟拉比法典》B.《论法的精神》C.《宅地法》D.《拿破仑法典》";
        List<JFastText.ProbLabel> probLabels = jft.predictProba(text, 5);
        for(JFastText.ProbLabel probLabel : probLabels){
            System.out.printf("\nThe label of '%s' is '%s' with probability %f\n",
                    text, probLabel.label, Math.exp(probLabel.logProb));
        }
        List<String> ps = jft.predict(text, 5);
        for(String s : ps){
            System.out.println(s);
        }
    }
}
